Skip to content

Use predict_df inside predict_fev#512

Merged
shchur merged 10 commits into
amazon-science:mainfrom
shchur:predict-fev-simplification
Jun 18, 2026
Merged

Use predict_df inside predict_fev#512
shchur merged 10 commits into
amazon-science:mainfrom
shchur:predict-fev-simplification

Conversation

@shchur

@shchur shchur commented Jun 17, 2026

Copy link
Copy Markdown
Contributor

Issue #, if available:

Description of changes:

  • Use predict_df + fev's built-in conversion utilities inside predict_fev for BaseChronosPipeline and Chronos2Pipeline.
  • Remove the bespoke fev-related logic in chronos.chronos2.dataset
  • Add batch_size support for BaseChronosPipeline.predict_df -> previously this method just ran inference on the entire DF in one go, potentially leading to OOM errors.

To do:

  • Run full benchmark for Chronos-2 and Chronos-Bolt to ensure that the results match v2.2.2

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@shchur shchur force-pushed the predict-fev-simplification branch from 090cb40 to 85cd9eb Compare June 17, 2026 12:37
@shchur shchur requested a review from abdulfatir June 17, 2026 14:39

@abdulfatir abdulfatir left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If the scores for Chronos-2 and Chronos-Bolt match, then this looks good to me. Thanks @shchur!

@shchur shchur merged commit 95185d1 into amazon-science:main Jun 18, 2026
10 of 11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants